from selenium import webdriver


def open_browser(browser='chrome'):
    """
    打开游览器，返回driver
    :param browser: str
    :return: driver
    """
    if browser == 'chrome':
        return webdriver.Chrome()
    elif browser == 'firefox':
        return webdriver.Firefox()
    elif browser == 'edge':
        return webdriver.Edge()
    else:
        print('请使用chrome，Firefox，edge这三种游览器')


if __name__ == '__main__':
    pass